How can I modify the Java code to group method calls by their class names